package com.chukun.flink.stream.partition;

import org.apache.flink.api.common.functions.Partitioner;

/**
 * @author chukun
 * @version 1.0.0
 * @description 简单分区策略
 * @createTime 2022年05月12日 00:11:00
 */
public class SimplePartitioner implements Partitioner<String> {
    @Override
    public int partition(String key, int i) {
        if (key.contains("185")) {
            return 0;
        } else if (key.contains("155")) {
            return 1;
        }
        return 2;
    }
}
